package com.project.ssm.controller;

import com.project.ssm.domain.SysLog;
import com.project.ssm.service.ISysLogService;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.context.SecurityContext;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.User;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.servlet.http.HttpServletRequest;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.Date;

@Component
@Aspect
public class LogAop {

    @Autowired
    private HttpServletRequest request;

    @Autowired
    private ISysLogService sysLogService;

    private Date visitTime; //开始时间
    private Class clazz;    //访问的类
    private Method method;  //访问的方法
    private String url;     //访问的url
    private String ip;      //访问的ip
    private String username;//当前访问用户

    //前置通知  主要时获取开始时间，执行的类是哪一个，执行的是哪一个方法
    @Before("execution(* com.project.ssm.controller.*.*(..))")
    public void doBefore(JoinPoint jp) throws NoSuchMethodException {
        visitTime = new Date();                             //当前时间就是开始时间
        clazz = jp.getTarget().getClass();                  //具体要访问的类
        String methodName = jp.getSignature().getName();    //获取访问的方法的名称
        Object[] args = jp.getArgs();                       //获取访问方法的参数

        //获取具体执行的方法的Method对象
        if(args == null || args.length == 0){
            method = clazz.getMethod(methodName);
        }else{
            Class[] classArgs = new Class[args.length];
            for(int i = 0; i < args.length; i++){
                classArgs[i] = args[i].getClass();
            }
            clazz.getMethod(methodName, classArgs);
        }
    }

    //后置通知
    @After("execution(* com.project.ssm.controller.*.*(..))")
    public void doAfter(JoinPoint jp) throws Exception {
        long time = new Date().getTime() - visitTime.getTime();

        //获取url
        if(clazz != null && method != null && clazz != SysLog.class){
            //获取类上的@RequestMapping的Value值
            RequestMapping classAnnotaion = (RequestMapping) clazz.getAnnotation(RequestMapping.class);
            if(classAnnotaion != null){
                String classValue = classAnnotaion.value()[0];
                //获取方法上的@RequestMapping的Value值
                RequestMapping methodAnnotation = (RequestMapping)method.getAnnotation(RequestMapping.class);
                if (methodAnnotation != null){
                    String methodValue = methodAnnotation.value()[0];
                    url = classValue + methodValue;
                    //获取ip
                    ip = request.getRemoteAddr();
                    //获取当前操作用户
                    SecurityContext context = SecurityContextHolder.getContext();
                    User user = (User)context.getAuthentication().getPrincipal();
                    username = user.getUsername();
                    SysLog sysLog = new SysLog();
                    sysLog.setExecutionTime(time);
                    sysLog.setIp(ip);
                    sysLog.setUrl(url);
                    sysLog.setUsername(username);
                    sysLog.setVisitTime(visitTime);
                    sysLog.setMethod("[类名]" + clazz.getName() + "\t[方法名]" + method.getName());
                    if(clazz.getName() != "com.project.ssm.controller.SysLogController"){
                        sysLogService.save(sysLog);
                    }
                }
            }
        }
    }
}
